import pandas as pd
import random

# 定义可能的姓名、职业和城市列表
names = ["张三", "李四", "王五", "赵六", "孙七", "周八", "吴九", "郑十", "王十一", "李十二"]
occupations = ["医生", "教师", "程序员", "厨师", "警察", "护士", "设计师", "演员", "作家", "司机"]
cities = ["北京", "上海", "广州", "深圳", "成都", "杭州", "武汉", "西安", "南京", "重庆"]

# 初始化空列表来存储每列的数据
data = {
    "姓名": [],
    "年龄": [],
    "性别": [],
    "职业": [],
    "所在城市": []
}

# 生成 20 行数据
for _ in range(20):
    data["姓名"].append(random.choice(names))
    data["年龄"].append(random.randint(18, 60))
    data["性别"].append(random.choice(["男", "女"]))
    data["职业"].append(random.choice(occupations))
    data["所在城市"].append(random.choice(cities))

# 创建 DataFrame
df = pd.DataFrame(data)

# 打印数据
print(df)

# 保存为 CSV 文件（可选）
df.to_csv("people_data.csv", index=False)